import org.junit.Test;

import java.io.File;

/**
 * @author: wuhao
 * @since: 2021/4/19 21:36
 */
public class FindJPG {

    //判断指定目录下是否有后缀名为.jpg的文件，如果有，就输出该文件名称

    @Test
    public void Test1(){
        String directory = "D:\\学习资料\\谷粒学院项目资料";
        String format = "jpg";
        File file = new File(directory);
        findJPG(file,format);
    }

    //找到此目录下所有的.jpg文件
    public void findJPG(File file, String format) {
         int count = 0;  //用于记录.jpg文件的数量
         //1.获取file目录下所有的File对象
        File[] files = file.listFiles();
        //2.遍历File数组
        for (int i = 0; i<files.length; i++){
            File fileOrDir = files[i];
            //3.判断是否是文件，是 比较格式
            if (fileOrDir.isFile()){
                String[] newString = fileOrDir.getName().split("\\.");
                if (format.equals(newString[1])){
                    count++;
                    System.out.println(fileOrDir.getName());
                }
            }
        }
        if (count==0){
            System.out.println("此文件目录下没有.jpg文件!");
        }
    }
}
